from ultralytics import YOLO
import os

os.environ["KMP_DUPLICATE_LIB_OK"] = "TRUE"

if __name__ == "__main__":
    # 事故
    # model_path = r"E:\项目\松山湖公安分局无人机自动巡检项目\事故检测\标注数据\OBB\权重文件\模型精度提升训练_样本数4947\weights\best.pt"
    # valid_yaml = r"E:\项目\松山湖公安分局无人机自动巡检项目\事故检测\标注数据\OBB\精度验证集\data.yaml"

    # 红外
    # model_path = r"E:\项目\松山湖公安分局无人机自动巡检项目\车辆检测\车辆识别标签\visdrone\模型训练权重\hugging_face_yolo_visdrone.pt"
    # model_path = r"E:\项目\松山湖公安分局无人机自动巡检项目\红外异常活动识别\标注\训练权重\HBB\样本_1763\train17\weights\best.pt"
    # model_path = r"C:\Users\14159\Desktop\其他工具\工具箱\ai\runs\detect\train10\weights\best.pt"
    # model_path = r"E:\zhihao\2025文档\25年9月\train\weights\best.pt"
    # valid_yaml=r"E:\zhihao\2025文档\25年9月\交通事故报告\第九周\data.yaml"
    # valid_yaml = r"E:\zhihao\2025文档\25年9月\train\精度验证集\data.yaml"

    # 红外推理
    # model_path = r"E:\zhihao\2025文档\25年9月\红外报告\weights\best.pt"
    # valid_yaml = r"E:\zhihao\2025文档\25年9月\红外报告\data.yaml"

    # 事故20250925
    # model_path = r"E:\zhihao\2025文档\25年9月\train\weights\best.pt"
    # model_path = r"E:\zhihao\2025文档\25年9月\ai\pt\20250925\train77\weights\best.pt"
    # valid_yaml = r"E:\zhihao\2025文档\25年9月\交通事故报告\原始数据\分集\验证\20250925\data.yaml"


    # 红外20250925
    # model_path = r"E:\zhihao\2025文档\25年9月\红外报告\weights\best.pt"
    # valid_yaml = r"E:\zhihao\2025文档\25年9月\红外报告\data.yaml"

    # 事故20251009
    # model_path = r"E:\zhihao\2025文档\25年10月\ai_训练\权重\20250925\train77\weights\best.pt"
    # valid_yaml = r"E:\zhihao\2025文档\25年10月\ai_训练\验证集\20251009\data.yaml"

    # 红外20251009
    model_path = r"E:\zhihao\2025文档\25年10月\ai_训练\权重\public_security\infrared_detection\20251009\weights\best.pt"
    valid_yaml = r"E:\zhihao\2025文档\25年10月\ai_训练\验证集\public_security\infrared_detection\20251009\data.yaml"

    # Load the model
    model = YOLO(model_path)

    # Run the evaluation
    # 验证精度
    results = model.val(data=valid_yaml, classes=[0], iou=0.3, batch=9)
    # 验证测试
    # results = model.predict(data=valid_yaml, classes=[0], iou=0.3, batch=9)
